JarDirectoryBuilder.java

/*
 *  Author:  Chris Seguin
 *
 *  This software has been developed under the copyleft
 *  rules of the GNU General Public License.  Please
 *  consult the GNU General Public License for more
 *  details about use and distribution of this software.
 */
package org.acm.seguin.tools.build;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.util.LinkedList;

import org.acm.seguin.io.DirectoryTreeTraversal;

/**
 *  Description of the Class 
 *
 *@author    Chris Seguin 
 */
public class JarDirectoryBuilder extends DirectoryTreeTraversal {
	private LinkedList directoryList;
	private PrintWriter output;
	private int rootLength;
	private boolean first;
	private String jarFile;
	private int directoryCount;
	private String manifest;

	private final static int MAX_DIRS = 5;


	/**
	 *  Constructor for the JarDirectoryBuilder object 
	 *
	 *@param  jarFile       Description of Parameter 
	 *@param  rootDir       Description of Parameter 
	 *@param  out           Description of Parameter 
	 *@param  initManifest  Description of Parameter 
	 */
	public JarDirectoryBuilder(String jarFile, String rootDir, PrintWriter out, String initManifest) {
		super(rootDir);
		directoryList = new LinkedList();
		output = out;
		first = true;
		rootLength = rootDir.length();
		this.jarFile = jarFile;
		directoryCount = 0;
		manifest = initManifest;
	}


	/**
	 *  Determines if this file should be handled by this traversal 
	 *
	 *@param  currentFile  the current file 
	 *@return              true if the file should be handled 
	 */
	protected boolean isTarget(File currentFile) {
		return currentFile.getName().endsWith(".class");
	}


	/**
	 *  Visits the current file 
	 *
	 *@param  currentFile  the current file 
	 */
	protected void visit(File currentFile) {
		String parentString = currentFile.getParent();
		//System.out.println("Parent:  " + parentString);
		if (!directoryList.contains(parentString)) {
			directoryList.add(parentString);

			if (directoryCount % MAX_DIRS == 0) {
				if (first) {
					output.print("jar cfm " + jarFile + " " + manifest);
					first = false;
				}
				else {
					output.print("jar uf " + jarFile);
				}
			}
			directoryCount++;

			if (parentString.length() == rootLength) {
				output.print(" *.class");
			}
			else {
				output.print(" " + parentString.substring(rootLength + 1) + 
						File.separator + "*.class");
			}

			if (directoryCount % MAX_DIRS == 0) {
				output.println("");
			}
		}
	}


	/**
	 *  The main program for the JarDirectoryBuilder class 
	 *
	 *@param  args  The command line arguments 
	 */
	public static void main(String[] args) {
		String jarFile = "sample.jar";
		String directory = System.getProperty("user.dir");
		PrintWriter out = null;
		String manifest = "META-INF/Manifest.MF";

		if (args.length > 0) {
			jarFile = args[0];
		}

		try {
			if (args.length > 1) {
				out = new PrintWriter(new FileWriter(args[1]));
			}
			else {
				out = new PrintWriter(new OutputStreamWriter(System.out));
			}
		}
		catch (IOException ioe) {
			System.out.println("Unable to create the output file:  " + args[0]);
			return;
		}

		if (args.length > 2) {
			directory = args[2];
		}

		if (args.length > 3) {
			manifest = args[3];
		}

		(new JarDirectoryBuilder(jarFile, directory, out, manifest)).go();

		out.println("");

		out.flush();
		out.close();
	}
}